#!/usr/bin env python
# -*- coding: utf-8 -*-
###########################################################################
#   Copyright (C) 2008 by Serj Poltavsky                                  #
#   serge.uliss at gmail.com                                              #
#                                                                         #
#   This program is free software; you can redistribute it and/or modify  #
#   it under the terms of the GNU General Public License as published by  #
#   the Free Software Foundation; either version 3 of the License, or     #
#   (at your option) any later version.                                   #
#                                                                         #
#   This program is distributed in the hope that it will be useful,       #
#   but WITHOUT ANY WARRANTY; without even the implied warranty of        #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         #
#   GNU General Public License for more details.                          #
#                                                                         #
#   You should have received a copy of the GNU General Public License     #
#   along with this program. If not, see <http://www.gnu.org/licenses/>   #
###########################################################################

import sys
from dvdfile import DvdFile

class DvdImage(DvdFile):
    _count = 0
    def __init__(self, filename, format):
        DvdFile.__init__(self, filename, format)
        DvdImage._count += 1
        self.count = DvdImage._count
        self.color = 0x000000

    def export(self):
        if self.filename is None:
            self.make_color_image(self.color)

    def make_color_image(self, color):
        from tools.image import ColorImage
        colorimage = ColorImage(color, self.format.size)
        colorimage.save(self.tmp_color_image(), "PNG")
        self.filename = self.tmp_color_image()

    def tmp_color_image(self):
        return '%scolorimage_%02i.png' %(self.tmp_dir(), self.count)